package com.leyou.lock;

import org.springframework.data.redis.core.StringRedisTemplate;

import java.util.concurrent.TimeUnit;

/**
 * redis分布式锁工具类
 */
public class RedisLock {
    private StringRedisTemplate redisTemplate;
    private String key;

    public RedisLock(StringRedisTemplate redisTemplate,String key){
            this.redisTemplate = redisTemplate;
            this.key = key;
    }

    /**
     * 锁对应的值，无意义，写为1
     */
    //private static final String value = "1";

    /**
     * 获取锁方法
     */
    public boolean lock(long locktime) {
        // 获取线程信息作为值，方便判断是否是自己的锁
        String value = Thread.currentThread().getId()+"";
        //setIfAbsent()方法底层就是发送setnx命令
        /**
         * setnx key value:
         如果该key不存在，则设置key和value到redis，返回1
         如果该key存在，无法设置key和value，返回0
         */
        // 尝试获取锁
        Boolean boo = redisTemplate.opsForValue().setIfAbsent(key, value, locktime, TimeUnit.SECONDS);
        // 判断结果
        return boo != null && boo;
    }

    /**
     * 释放锁方法
     */
    public void unlock() {
        // 获取线程信息作为值，方便判断是否是自己的锁
        String value =  Thread.currentThread().getId()+"";
        // 获取现在的锁的值
        String val = redisTemplate.opsForValue().get(key);
        // 判断是否是自己
        if(value.equals(val)) {
            // 删除key即可释放锁
            redisTemplate.delete(key);
        }
    }

}
